
#include "register/tilingdata_base.h"

namespace optiling {
BEGIN_TILING_DATA_DEF(ArgMaxWithValueTilingData)
  TILING_DATA_FIELD_DEF(uint32_t, magicNum);
  TILING_DATA_FIELD_DEF(uint32_t, axesDim);
  TILING_DATA_FIELD_DEF(uint32_t, smallDataNum);
  TILING_DATA_FIELD_DEF(uint32_t, tailBlockNum);
  TILING_DATA_FIELD_DEF(uint32_t, totalLength);
  TILING_DATA_FIELD_DEF(uint32_t, dims);
  TILING_DATA_FIELD_DEF(uint32_t, stride);
  TILING_DATA_FIELD_DEF(uint32_t, nIter);
  TILING_DATA_FIELD_DEF(uint32_t, iterStep);
  TILING_DATA_FIELD_DEF(uint32_t, resLength);
  TILING_DATA_FIELD_DEF(uint32_t, bigDataNum);
  TILING_DATA_FIELD_DEF(int64_t, axes);
  TILING_DATA_FIELD_DEF(bool, keep_dims);
END_TILING_DATA_DEF;

REGISTER_TILING_DATA_CLASS(ArgMaxWithValue, ArgMaxWithValueTilingData)
}
